how to use filter function in Python